While loop in Python